// include/add_custom_tiling.h
#ifndef ADD_CUSTOM_TILING_H
#define ADD_CUSTOM_TILING_H

#include <cstdint>

namespace gert {
    class TilingContext;
    class Shape;
    class InferShapeContext;
    class InferDataTypeContext;
    class RawTilingData;
}

namespace ge {
    enum DataType;
    enum Format;
    enum ParamType;
    class graphStatus;
}

class TilingData {
public:
    void set_totalLength(uint32_t length);
    void set_tileNum(uint32_t num);
    size_t GetDataSize() const;
    void SaveToBuffer(void* buffer, size_t capacity) const;
};

#endif // ADD_CUSTOM_TILING_H